package com.ylw.util;

import org.apache.poi.xwpf.usermodel.XWPFTable;
import org.apache.poi.xwpf.usermodel.XWPFTableCell;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTDecimalNumber;
import org.openxmlformats.schemas.wordprocessingml.x2006.main.CTTcPr;

import java.math.BigInteger;
import java.util.ArrayList;
import java.util.List;

public class XWPFTableUtil {



    /**
     * 获取当前单元格的rowspan（行合并）的行数
     *
     * @param table 表格
     * @param row 行值
     * @param col 列值
     * @return
     */
    public static int getRowspan(XWPFTable table, int row, int col) {

        XWPFTableCell cell = table.getRow(row).getCell(col);
        // 正常独立单元格
        if (!isContinueRow(cell) && !isRestartRow(cell)) {
            return 1;
        }
        // 当前单元格的宽度
        int cellWidth = getCellWidth(table, row, col);
        // 当前单元格距离左侧边框的距离
        int leftWidth = getLeftWidth(table, row, col);

        // 用户保存当前单元格行合并的单元格数-1（因为不包含自身）
        List<Boolean> list = new ArrayList();
        getRowspan(table, row, cellWidth, leftWidth, list);

        return list.size() + 1;
    }

    private static void getRowspan(XWPFTable table, int row, int cellWidth, int leftWidth,
                            List<Boolean> list) {
        // 已达到最后一行
        if (row + 1 >= table.getNumberOfRows()) {
            return;
        }
        row = row + 1;
        int colsNum = table.getRow(row).getTableCells().size();
        // 因为列合并单元格可能导致行合并的单元格并不在同一列，所以从头遍历列，通过属性、宽度以及距离左边框间距来判断是否是行合并
        for (int i = 0; i < colsNum; i++) {
            XWPFTableCell testTable = table.getRow(row).getCell(i);
            // 是否为合并单元格的中间行（包括结尾行）
            if (isContinueRow(testTable)) {
                // 是被上一行单元格合并的单元格
                if (getCellWidth(table, row, i) == cellWidth
                        && getLeftWidth(table, row, i) == leftWidth) {
                    list.add(true);
                    // 去下一行继续查找
                    getRowspan(table, row, cellWidth, leftWidth, list);
                    break;
                }
            }
        }
    }



    /**
     * 判断是否是合并行的中间行单元格（包括结尾的最后一行的单元格）
     *
        打印Tcpt正常单元格为
         <w:tcW w:type="dxa" w:w="1220"/>
         <w:tcBorders>
         <w:top w:color="auto" w:space="0" w:sz="4" w:val="single"/>
         <w:left w:color="auto" w:space="0" w:sz="4" w:val="single"/>
         <w:bottom w:color="auto" w:space="0" w:sz="4" w:val="single"/>
         <w:right w:color="auto" w:space="0" w:sz="4" w:val="single"/>
         </w:tcBorders>
         <w:hideMark/>
     出现行合并的单元格为
         <w:tcW w:type="dxa" w:w="1220"/>
         <w:vMerge w:val="restart"/>           起始合并行
         <w:tcBorders>
         <w:top w:color="auto" w:space="0" w:sz="4" w:val="single"/>
         <w:left w:color="auto" w:space="0" w:sz="4" w:val="single"/>
         <w:right w:color="auto" w:space="0" w:sz="4" w:val="single"/>
         </w:tcBorders>
         <w:hideMark/>

        被合并行
         <w:tcW w:type="dxa" w:w="1220"/>
         <w:vMerge/>
         <w:tcBorders>
         <w:left w:color="auto" w:space="0" w:sz="4" w:val="single"/>
         <w:right w:color="auto" w:space="0" w:sz="4" w:val="single"/>
         </w:tcBorders>
         <w:hideMark/>
     * @param tableCell
     * @return
     */
    public static boolean isContinueRow(XWPFTableCell tableCell) {
        CTTcPr tcPr = tableCell.getCTTc().getTcPr();
        if (tcPr.getVMerge() == null) {
            return false;
        }
        if (tcPr.getVMerge().getVal() == null) {
            return true;
        }
        return false;
    }

    /**
     * 判断是否是合并行的起始行单元格
     *
     * @param tableCell
     * @return
     */
    public static boolean isRestartRow(XWPFTableCell tableCell) {
        CTTcPr tcPr = tableCell.getCTTc().getTcPr();
        if (tcPr.getVMerge() == null) {
            return false;
        }
        if (tcPr.getVMerge().getVal() == null) {
            return false;
        }
        if (tcPr.getVMerge().getVal().toString().equalsIgnoreCase("restart")) {
            return true;
        }
        return false;
    }


    public static int getLeftWidth(XWPFTable table, int row, int col) {
        int leftWidth = 0;
        for (int i = 0; i < col; i++) {
            leftWidth += getCellWidth(table, row, i);
        }
        return leftWidth;
    }

    public static int getCellWidth(XWPFTable table, int row, int col) {
        BigInteger width = table.getRow(row).getCell(col).getCTTc().getTcPr().getTcW().getW();
        return width.intValue();
    }


    /**
     * 获取当前单元格的colspan（列合并）的列数
     *
     * @param tcPr 单元格属性
     * @return
     */
    public static int getColspan(CTTcPr tcPr) {
//         判断是否存在列合并
        CTDecimalNumber gridSpan = null;
        if ((gridSpan = tcPr.getGridSpan()) != null) { // 合并的起始列
            // 获取合并的列数
            BigInteger num = gridSpan.getVal();
            return num.intValue();
        } else { // 其他被合并的列或正常列
            return 1;
        }
    }
}
